﻿#include "pch.h"
#include "framework.h"
#include "yyAccounting.h"
#include "yyAccountingDlg.h"
#include "afxdialogex.h"
#include "FinanceEntry.h"
#include <fstream>
#include <sstream>
#include "common.h"
#include "SqliteDB.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

// 用于应用程序“关于”菜单项的 CAboutDlg 对话框
class CAboutDlg : public CDialogEx {
public:
    CAboutDlg();

// 对话框数据
#ifdef AFX_DESIGN_TIME
    enum { IDD = IDD_ABOUTBOX };
#endif

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持

// 实现
protected:
    DECLARE_MESSAGE_MAP()
};

CAboutDlg::CAboutDlg() : CDialogEx(IDD_ABOUTBOX) {
}

void CAboutDlg::DoDataExchange(CDataExchange* pDX) {
    CDialogEx::DoDataExchange(pDX);
}

BEGIN_MESSAGE_MAP(CAboutDlg, CDialogEx)
END_MESSAGE_MAP()

// --------------------------------------------------------------分割线-------------------------------------------------------------
// CyyAccountingDlg 对话框
CyyAccountingDlg::CyyAccountingDlg(CWnd* pParent /*=nullptr*/)
    : CDialogEx(IDD_YYACCOUNTING_DIALOG, pParent), m_db("./yyAccounting.db") {
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CyyAccountingDlg::DoDataExchange(CDataExchange* pDX) {
    CDialogEx::DoDataExchange(pDX);
    DDX_Control(pDX, IDC_COMBO1, m_cbx1);
    DDX_Control(pDX, IDC_ITEM_LIST, m_ItemList);
}

BEGIN_MESSAGE_MAP(CyyAccountingDlg, CDialogEx)
    ON_WM_SYSCOMMAND()
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_BN_CLICKED(IDC_BUTTON1, &CyyAccountingDlg::OnBnClickedButton1)
END_MESSAGE_MAP()

// 初始化账单种类下拉框
void CyyAccountingDlg::InitComboBox() {
    std::fstream fs("./entry_types.csv", std::ios::in);
    if (!fs.is_open()) {
        MessageBox(TEXT("无法读取配置文件!"));
        exit(0);
    }

    std::string line;
    while (std::getline(fs, line)) {
        all_entry_types.emplace_back(line);
    }
    fs.close();

    for (size_t i = 0; i < all_entry_types.size(); i++) {
        m_cbx1.AddString((CString)all_entry_types[i].c_str());
    }

    m_cbx1.SetCurSel(0);    // 设置默认选项为 "请选择记账类型"
}

// 初始化收入或支出按钮
void CyyAccountingDlg::InitListControl() {
    CRect rect;
    m_ItemList.GetClientRect(&rect);    // 获取列表视图控件的位置和大小

    m_ItemList.SetExtendedStyle(m_ItemList.GetExtendedStyle() | LVS_EX_FULLROWSELECT | LVS_EX_GRIDLINES);   // 为列表视图控件添加全行选中和栅格风格

    // 为列表视图控件添加五列
    m_ItemList.InsertColumn(0, _T("id"), LVCFMT_CENTER, rect.Width() / 5, 0);
    m_ItemList.InsertColumn(1, _T("time"), LVCFMT_CENTER, rect.Width() / 5, 0);
    m_ItemList.InsertColumn(2, _T("income"), LVCFMT_CENTER, rect.Width() / 5, 0);
    m_ItemList.InsertColumn(3, _T("entry_type_str"), LVCFMT_CENTER, rect.Width() / 5, 0);
    m_ItemList.InsertColumn(4, _T("money"), LVCFMT_CENTER, rect.Width() / 5, 0);
}

BOOL CyyAccountingDlg::OnInitDialog() {
    CDialogEx::OnInitDialog();
    // 将“关于...”菜单项添加到系统菜单中。
    // IDM_ABOUTBOX 必须在系统命令范围内。
    ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
    ASSERT(IDM_ABOUTBOX < 0xF000);

    CMenu* pSysMenu = GetSystemMenu(FALSE);
    if (pSysMenu != nullptr) {
        BOOL bNameValid;
        CString strAboutMenu;
        bNameValid = strAboutMenu.LoadString(IDS_ABOUTBOX);
        ASSERT(bNameValid);
        if (!strAboutMenu.IsEmpty()) {
            pSysMenu->AppendMenu(MF_SEPARATOR);
            pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
        }
    }

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);     // 设置大图标
    SetIcon(m_hIcon, FALSE);    // 设置小图标

    // TODO: 在此添加额外的初始化代码
    // ((CButton*)GetDlgItem(IDC_RADIOMan))->SetCheck(true)  默认选中按钮
    InitComboBox();     // 初始化下拉框
    InitListControl();  // 初始化表格

    return TRUE;        // 除非将焦点设置到控件，否则返回 TRUE
}

void CyyAccountingDlg::OnSysCommand(UINT nID, LPARAM lParam) {
    if ((nID & 0xFFF0) == IDM_ABOUTBOX) {
        CAboutDlg dlgAbout;
        dlgAbout.DoModal();
    } else {
        CDialogEx::OnSysCommand(nID, lParam);
    }
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。
void CyyAccountingDlg::OnPaint() {
    if (IsIconic()) {
        CPaintDC dc(this); // 用于绘制的设备上下文

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), 0);

        // 使图标在工作区矩形中居中
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + 1) / 2;
        int y = (rect.Height() - cyIcon + 1) / 2;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    } else {
        CDialogEx::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CyyAccountingDlg::OnQueryDragIcon() {
    return static_cast<HCURSOR>(m_hIcon);
}

// 展示最近条目
void CyyAccountingDlg::ShowInListControl() {
    m_ItemList.DeleteAllItems();    // 清空上次数据
    read_ret data;
    m_db.ReadRecentItems(16, data);
    int index = 1;
    for (int i=0;i<data.size();i++) {
        index = 1;
        m_ItemList.InsertItem(i, CString(std::to_string(data[i].GetId()).c_str()));
        m_ItemList.SetItemText(i, index++, CString(data[i].GetTime().c_str()));
        m_ItemList.SetItemText(i, index++, CString(std::to_string(data[i].GetIncome()).c_str()));
        m_ItemList.SetItemText(i, index++, CString(data[i].GetEntryTypeStr().c_str()));
        m_ItemList.SetItemText(i, index++, CString(std::to_string(data[i].GetMoney()).c_str()));
    }
}

// 记账按钮
void CyyAccountingDlg::OnBnClickedButton1() {
    CTime date;
    ((CDateTimeCtrl*)GetDlgItem(IDC_DATETIMEPICKER1))->GetTime(date);
    bool income = ((CButton*)GetDlgItem(IDC_RADIO1))->GetCheck();
    CString money;
    GetDlgItemText(IDC_EDIT1, money);
    int curIndex = m_cbx1.GetCurSel();

    CFinanceEntry financeEntry = CFinanceEntry((LPCSTR)(CStringA)(date.Format("%Y-%m-%d")), income, curIndex, all_entry_types[curIndex], _ttof(money));

    if (!m_db.WriteToDB(financeEntry)) {
        MessageBox(TEXT("write fail!"));
    }

    ShowInListControl();

    return;
}


